# -*- coding: utf-8 -*-
"""
    @Author：SimpleTree
    @date：2025/6/5 11:14
    @desc:
"""
from typing import List

from django.db.models import QuerySet

from embedding.models import Embedding
from embedding.vector.base_vector import BaseVectorStore


class PGVector(BaseVectorStore):

    def delete_by_dataset_id_list(self, dataset_id_list: List[str]):
        QuerySet(Embedding).filter(dataset_id__in=dataset_id_list).delete()
